package org.example.Greedy_Algorithm;

import java.util.Arrays;
import java.util.Comparator;

public class Reconstruct_the_queue_based_on_height {
    public static void main(String[] args) {
        //根据身高重建队列

        //假设有打乱顺序的一群人站成一个队列，数组 people 表示队列中一些人的属性（不一定按顺序）。每个 people{i] = [hi, ki] 表示第 i 个人的身高为 hi ，前面 正好 有 ki 个身高大于或等于 hi 的人。
        //请你重新构造并返回输入数组 people 所表示的队列。返回的队列应该格式化为数组 queue ，其中 queue[j] = [hj, kj] 是队列中第 j 个人的属性（queue[0] 是排在队列前面的人）。
        int[][] people = {{6,0},{5,0},{4,0},{3,2},{2,2},{1,4}};
        int[][] queue = reconstructQueue(people);
        System.out.println(Arrays.deepToString(queue));
    }
    public static int[][] reconstructQueue(int[][] people) {
        Arrays.sort(people, new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                int com = Integer.compare(o1[1], o2[1]);
                if (com==0){
                    return Integer.compare(o1[0], o2[0]);
                }
                return com;
            }
        });
        System.out.println(Arrays.deepToString(people));
        for (int i = 0; i < people.length; i++) {
            int num = 0;
            for (int j = 0; j < i; j++) {
                if (people[j][0]>=people[i][0]){
                    num++;
                }
            }
            if (num!=people[i][1]){
                int step = people[i][1] - num;
                if (step<0) step = -step;
                int pre = i-1;
                int[] temp = people[i];
                while(step>0){
                    people[pre+1] = people[pre];
                    pre--;
                    step--;
                }
                people[pre+1] = temp;
            }
        }
        return people;
    }
}
